from config.embedding_config import get_openai_embeddings_local

if __name__ == '__main__':
    ollamEmbedding = get_openai_embeddings_local()
    r1 = ollamEmbedding.embed_documents(["梦里花落知多少作者是谁？"])
    r2 = ollamEmbedding.embed_documents(["《梦里花落知多少》是当代作家郭敬明创作的一部长篇小说，于 2004 年出版。"
                                         "小说以北京、上海等大都市为背景，讲述了几个年轻人的爱情故事，故事情节曲折，"
                                         "语言幽默生动，大量的引入口语与通俗幽默的比喻。"])
    r3 = ollamEmbedding.embed_documents(["我今天吃了一次非常好吃的火锅？这盘火锅是王老师请的，王老师真是一个好人，"
                                         "请我们吃这么好吃的火锅，还记得小时候，我们在老家和王老师一起的时候，"
                                         "他就总是请我们去他家里吃饭，那时缺衣少食，都是王老师让我们吃好长好的。"])

    print("----similarity-----")
    print(self.calcCosSimilarity(r1, r2))
    print("----similarity2-----")
    print(self.calcCosSimilarity(r1, r3))